import pandas as pd

# 定义LSR计算函数
def calculate_LSR(row):
    control_nature = row['实际控制人性质'].split(',')[0]  # 获取第一个值
    if control_nature in ['1100', '2000', '2100', '2120']:
        return row['短期负债'] + 0.5 * row['长期负债']  # 国有企业
    else:
        return row['短期负债'] + 0.4 * row['长期负债']  # 非国有企业

# 读取a2.xlsx文件
file_path = 'a2.xlsx'
df = pd.read_excel(file_path)

# 确保必要的列存在
required_columns = ['实际控制人性质', '短期负债', '长期负债', '年份']
missing_columns = [col for col in required_columns if col not in df.columns]
if missing_columns:
    raise ValueError(f"Excel文件中缺少以下列：{missing_columns}")

# 添加新列“所有权”
df['所有权'] = ''

# 填充“所有权”列
df['所有权'] = df['实际控制人性质'].apply(lambda x: '国有企业' if x.split(',')[0] in ['1100', '2000', '2100', '2120'] else '非国有企业')

# 计算LSR值
df['LSR'] = df.apply(calculate_LSR, axis=1)

# 对相同年份的LSR值进行均值法处理
grouped_df = df.groupby(['年份', '所有权'])['LSR'].mean().reset_index()

# 将处理后的数据保存到新的Excel文件b1.xlsx中
output_file_path = 'b1.xlsx'
grouped_df.to_excel(output_file_path, index=False)

print(f"均值法处理后的LSR值已保存到 '{output_file_path}' 文件中。")
